#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
模型包
"""

from .i3d import I3D
from .fusion import SpatioTemporalFusion
from .videomae import VideoMAE
from .swin_transformer_3d import SwinTransformer3D

__all__ = ['I3D', 'SpatioTemporalFusion', 'VideoMAE', 'SwinTransformer3D'] 